import cupy as cp

# 检查CuPy是否在使用GPU
print("CuPy using GPU:", cp.cuda.is_available())
